90b630fa28dc6ec184e27193906c316c62212368,h2o-algos/src/test/java/hex/deeplearning/DeepLearningIrisTest.java,DeepLearningIrisTest,runFraction,#number#,30
Before Change
// implementation has hardcoded). But count
// of classes is not known unless we visit
// all the response data - force that now.
_train.replace(_train.numCols()-1,_train.lastVec().toEnum());
_test .replace(_train.numCols()-1,_test .lastVec().toEnum());
DKV.put(_train._key,_train);
DKV.put(_test ._key,_test );
After Change
// of classes is not known unless we visit
// all the response data - force that now.
String respname = _train.lastVecName();
Vec resp = _train.lastVec().toEnum();
_train.remove(respname).remove();
_train.add(respname, resp);
Vec vresp = _test.lastVec().toEnum();
_test.remove(respname).remove();